
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <strings.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <net/if.h>
#include <sys/ioctl.h>
#include <errno.h>
#include <fcntl.h>
#include <poll.h>

// 服务端，被动接收连接请求

int main(int argc, char const *argv[]) // ./server PORT
{
    if(argc != 2)
    {
        printf("请指定绑定端口\n");
        exit(0);
    }

    // 1，创建TCP
    int sockfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

    // 2，准备好服务器的地址
    struct sockaddr_in addr;
    socklen_t len = sizeof(addr);
    bzero(&addr, len);
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = htonl(INADDR_ANY);
    addr.sin_port = htons(atoi(argv[1]));

    // 在实验测试阶段  重组
    // 避免地址资源IP:PORT在程序退出之后（即套接字关闭之后）被系统保留
    int on = 1;
    setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));

    // 3，绑定sockfd和地址
    if(bind(sockfd, (struct sockaddr *)&addr, len) == -1)
    {
        perror("绑定地址失败");
        exit(0);
    }

    // 4，设定监听状态
    // 第二个参数指的是同时接收连接请求的个数
    listen(sockfd, 3);

    // 5，静静地等待客户端的请求
    struct sockaddr_in peeraddr;
    len = sizeof(peeraddr);
    bzero(&peeraddr, len);
    int connfd = accept(sockfd, (struct sockaddr *)&peeraddr, &len);
    if(connfd > 0)
    {
        printf("有新的客户端连接：[%s:%hu]\n",
                inet_ntoa(peeraddr.sin_addr),
                ntohs(peeraddr.sin_port));
    }


    // 6，聊天
    //    同时监控connfd、STDIN_FILENO
    char buf[100];

    struct pollfd fds[2];
    bzero(&fds, sizeof(fds));
    fds[0].fd = connfd;
    fds[0].events = POLLIN|POLLPRI;
    fds[1].fd = STDIN_FILENO;
    fds[1].events = POLLIN;

    while(1)
    {
        if(poll(fds, 2, -1) == -1)
        {
            perror("poll() failed");
            break;
        }

        if(fds[0].revents & POLLIN) 
        {
            read(connfd, buf, 100);
            printf("收到TCP普通数据：%s", buf);
        }

        if(fds[0].revents & POLLPRI)
        {
            char oob;
            recv(connfd, &oob, 1, MSG_OOB);
            printf("收到TCP紧急数据：%c", oob);
        }

        if(fds[1].revents & POLLIN)
        {
            bzero(buf, 100);
            fgets(buf, 100, stdin);
            write(connfd, buf, strlen(buf));
        }
    }

    return 0;
}